package com.example.analysis.utils;


import com.example.analysis.entity.Evaluate2;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * excel文件的工具类
 * 有几类excel文件
 * 学生成绩
 *  评教数据
 *  竞赛信息
 */
public class ExcelUtils4 {
    //总行数
    private static int totalRows = 0;
    //总条数
    private static int totalCells = 0;
    //错误信息接收器
    private static String errorMsg;

    public static List<Evaluate2> getExcelInfo(MultipartFile mFile){
        String fileName = mFile.getOriginalFilename();//获取文件名
        try {
            if (!validateExcel(fileName)) {// 验证文件名是否合格
                return null;
            }
            boolean isExcel2003 = true;// 根据文件名判断文件是2003版本还是2007版本
            if (isExcel2007(fileName)) {
                isExcel2003 = false;
            }
            List<Evaluate2>  Evaluate2List = createExcel(mFile.getInputStream(), isExcel2003);
            return Evaluate2List;

        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
    /**
     * 不同的excel版本
     * @param is 输入流
     * @param isExcel2003 excel是2003还是2007版本
     * @return
     * @throws IOException
     */
    public static List<Evaluate2> createExcel(InputStream is, boolean isExcel2003) {
        try{
            Workbook wb = null;
            if (isExcel2003) {// 当excel是2003时,创建excel2003
                wb = new HSSFWorkbook(is);
            } else {// 当excel是2007时,创建excel2007
                wb = new XSSFWorkbook(is);
            }
            List<Evaluate2> Evaluate2List = readExcelValue(wb);// 读取Excel里面客户的信息
            return Evaluate2List;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
    /**
     * 读取Excel文件
     */
    public static List<Evaluate2> readExcelValue(Workbook wb) {
        //默认会跳过第一行标题
        // 得到第一个shell
        Sheet sheet = wb.getSheetAt(0);
        // 得到Excel的行数
        totalRows = sheet.getPhysicalNumberOfRows();
        // 得到Excel的列数(前提是有行数)
        if (totalRows > 1 && sheet.getRow(0) != null) {
            totalCells = sheet.getRow(2).getPhysicalNumberOfCells();
        }
        List<Evaluate2> Evaluate2List = new ArrayList<Evaluate2>();
        // 循环Excel行数  第二行是表头
        for (int r = 3; r < totalRows - 1; r++) {
            Row row = sheet.getRow(r);
            if (row == null) {
                continue;
            }
            Evaluate2 aEvaluate2 = new Evaluate2();
            // 循环Excel的列
            for (int c = 0; c < totalCells - 1 ; c++) {
                int[] temp=new int[10];

                Cell cell = row.getCell(c);
                if (sheet.getRow(2).getCell(c).getStringCellValue().equals("教师账号")&& cell!=null){
                    aEvaluate2.setTeacherId(cell.getStringCellValue());
                }
                else if (sheet.getRow(2).getCell(c).getStringCellValue().equals("授课教师")&& cell!=null){
                    aEvaluate2.setTeacherName(cell.getStringCellValue());
                }
                else if (sheet.getRow(2).getCell(c).getStringCellValue().equals("课程编号")&& cell!=null){
                    aEvaluate2.setCourseId(cell.getStringCellValue());
                }
                else if (sheet.getRow(2).getCell(c).getStringCellValue().equals("课程名称")&& cell!=null){
                    aEvaluate2.setCourseName(cell.getStringCellValue());
                }
                else if (sheet.getRow(2).getCell(c).getStringCellValue().equals("听课人")&& cell!=null){
                    aEvaluate2.setSupervision(cell.getStringCellValue());
                }
                else if (sheet.getRow(2).getCell(c).getStringCellValue().equals("上课时间")&& cell!=null){
                    aEvaluate2.setEvaTime(cell.getStringCellValue());
                }
                else if (c==44 && r!=3 && !cell.getStringCellValue().isEmpty()){
                    aEvaluate2.setQuest1(Double.valueOf(cell.getStringCellValue()));
                }
                else if (c==45 && r!=3 && !cell.getStringCellValue().isEmpty()){
                    aEvaluate2.setQuest2(Double.valueOf(cell.getStringCellValue()));
                }
                else if (c==46 && r!=3 && !cell.getStringCellValue().isEmpty()){
                    aEvaluate2.setQuest3(Double.valueOf(cell.getStringCellValue()));
                }
                else if (c==47 && r!=3 && !cell.getStringCellValue().isEmpty() ){
                    aEvaluate2.setQuest4(Double.valueOf(cell.getStringCellValue()));
                }
                else if (c==48 && r!=3 && !cell.getStringCellValue().isEmpty() ){
                    aEvaluate2.setQuest5(Double.valueOf(cell.getStringCellValue()));
                }
                else if (c==49 && r!=3 && !cell.getStringCellValue().isEmpty() ){
                    aEvaluate2.setQuest6(Double.valueOf(cell.getStringCellValue()));
                }
            }
            Evaluate2List.add(aEvaluate2);
        }
        System.out.println("文件数量：");
        System.out.println(totalRows + "   " + totalCells);
        return Evaluate2List;
    }


    public static boolean validateExcel(String filePath) {
        if (filePath == null || !(isExcel2003(filePath) || isExcel2007(filePath))) {
            errorMsg = "文件名不是excel格式";
            return false;
        }
        return true;
    }
    // @描述：是否是2003的excel，返回true是2003
    public static boolean isExcel2003(String filePath)  {
        return filePath.matches("^.+\\.(?i)(xls)$");
    }
    //@描述：是否是2007的excel，返回true是2007
    public static boolean isExcel2007(String filePath)  {
        return filePath.matches("^.+\\.(?i)(xlsx)$");
    }

}
